cd ../..

gpus=$1

# python joint_main.py \
#     --gpus ${gpus}, \
#     --opt_conf ./configs/opt_joint.yaml \
#     --exp_name joint/OOD_CRNN_w1.0_wMT_oRNN \
#     --feat_dim 256 \
#     --grad_compensate False \
#     --joint_itpl_exponent 0 \
#     --joint_itpl_warmup 80 \
#     --ood_weight 1.0

# Initialize CNN version
for freeze_layer in 3
do
    echo "Freeze CNN layer ${freeze_layer}"
    python joint_main.py \
        --gpus ${gpus}, \
        --opt_conf ./configs/opt_joint.yaml \
        --exp_name joint/OOD_CRNN_w1.0_wMT_o${freeze_layer}CNN+RNN \
        --feat_dim 256 \
        --grad_compensate False \
        --joint_itpl_exponent 0 \
        --joint_itpl_warmup 80 \
        --ood_weight 1.0 \
        --batch_sizes '[0, 12, 12, 24, 96]' \
        --self_usage '[0, 0, 0, 1, 1]' \
        --freeze_cnn_layer '[0, 1, 2]'
done